Each Reward to Its Relevant Tokens: Reward-Specific Token-Level Credit Assignment for Multi-Reward RL
多奖励 RL 的逐奖励 Token 级信用分配
🧭 多奖励 RL 令牌级信用分配 ★ 深度解读
University of Minnesota · National University of Singapore · University of Illinois Urbana-Champaign · University of Southern California · University of Pennsylvania
用 verifier 信息 self-replay 定位每个 reward 的相关 token,把 advantage 按相关性重分配,加速并提升多奖励 RL
📌 工作参考价值(4 点)
- 对电商/Agent 里带多个可验证约束的生成任务(商品文案、广告标题、工具调用、SQL/API 生成),把每个 reward 的 advantage 按 self-replay 的相关性分配到 token,而不是整段均匀;尤其格式类 reward 只落在标签/keyword/CTA 等少数 token,能显著加快收敛。
- 若 verifier 有可文本化信息(参考 query、执行 error、约束条件),可直接复用 MRTPO 的 leave-one-out replay:同一条 response 在 P_all 和 P_\k 各 forward 一次,用 next-token 分布的 JS 距离得到 token 权重;无需额外模型或标注,成本只比 GDPO 多 10–20%,且只在 reward 组内有方差时 replay。
- β=0.5 是不错的默认值,前 10 步 smoothstep warmup;保留 1−β 的均匀份额避免重分配过激,并可对 reallocated part 做二阶矩约束。这个方法正交于 SAW/DVAO 等 response 级 reward reweighting,可以叠加。
- 做 prompt 约束(如“必须包含关键词、以某句结尾、加双引号”)时,直接去掉单个约束 replay 能定位到 checker 实际读取的 token;对 IFBench 这类未见约束泛化提升最明显,适合冷启动或模板频繁变化的场景。